Dense Structural Compression of Transformers via Gauge-Correct Channel Removal
通过规范校正通道移除实现 Transformer 稠密结构化压缩
🧭 训练期结构化压缩与架构搜索 ★ 深度解读
Université d’Artois, CNRS, CRIL UMR 8188, Lens, France
训练期用对称 group-lasso 惩罚解决 gauge 自由度,在保留 dense tensor 下动态压缩 Transformer,最高 255 倍 FMA 压缩且精度不降
📌 工作参考价值(4 点)
- 部署推荐/广告 Transformer 时,可优先考虑训练期自适应压缩而非后剪枝:论文证明同样的 utility 度量在后剪枝上失败或显著变差,持续压缩压力会促使网络学到更高效的表征,而不仅删除冗余。
- 如果业务模型含 SwiGLU MLP、attention 等 gauge-connected 结构,可借鉴 additive symmetric group-lasso 惩罚:同时对两侧 factor 做 Frobenius-norm 惩罚,避免直接惩罚 product norm 导致的梯度消失和不稳定;按每通道实际 FMA/参数规模校准 penalty,能把「效用/算力」统一到同一尺度。
- 工程上实现物理 channel removal 而非 mask 稀疏:用 gauge Sinkhorn 先平衡、再 trust-drop 移除整 slice,缩小后的 dense tensor 在 GPU 上保持高吞吐,减少稀疏算子低效;这适合做粗粒度 tower 压缩或小规模重训模型。
- 自适应 ρ 由 loss target 驱动压缩强度,实现较平滑的「先达标再压缩」过程,可作为业务中自动搜架构、控制性能下限的参考;但该理论目前只在 <150M FMA 从零训练任务上验证,迁移到大模型或预训练权重需谨慎。